from common.basePage import Base
from page.page_devicemodel import DeviceModel
from page.page_login import Login
from page.page_service_rule import ServiceRule


class Index(Base):
    # 封装所有page类并返回

    def index_login(self):
        return Login(self.driver)

    def index_devicemodel(self):
        return DeviceModel(self.driver)

    def index_service_rule(self):
        return ServiceRule(self.driver)

index = Index()